// #ifndef _THREAD_HPP__
// #define _THREAD_HPP__
#pragma once

#include <iostream>
#include <thread>
#include <string>
#include <functional>
#include <unistd.h>
#include <sys/types.h>

namespace ThreadModule
{
    using func_t = std::function<void()>; // func_t 之后就是一个函数类型
    static int number = 1;

    enum class TSTATUS // 枚举类线程的状态
    {
        NEW,     // 新线程
        RUNNING, // 正在运行
        STOP     // 停止
    };

    class Thread
    {
    private:
        // 成员方法！！！！有this指针 有2个参数 所以下面Routine报错       两种方法：1.把this指针变成静态成员 1.把Routine放在类外(不建议)
        static void *Routine(void *args)
        {
            Thread *t = static_cast<Thread *>(args); // 把void*转成Thread*  拿到当前对象的this指针了
            t->_status = TSTATUS::RUNNING;
            t->_func(); // 调用类内成员 但是静态成员没有this指针
            return nullptr;
        }
        void EnableDetach() // 让线程开启可以分离
        {
            _joinable = false;
        }

    public:
        Thread(func_t func) : _func(func), _status(TSTATUS::NEW), _joinable(true)
        {
            _name = "Thread-" + std::to_string(number++);
            _pid = getpid(); // 获取线程的pid
        }
        bool Start()
        {
            if (_status != TSTATUS::RUNNING)
            {
                int n = ::pthread_create(&_tid, nullptr, Routine, this);
                if (n != 0)
                    return false;
                return true;
            }
            return false;
        }
        bool Stop()
        {
            if (_status == TSTATUS::RUNNING) // 只有线程在RUNNING状态才需要Stop
            {
                // 直接join就退出了 要想stop 就先stop再join
                int n = ::pthread_cancel(_tid);
                if (n != 0)
                    return false;
                _status = TSTATUS::STOP;
                return true;
            }
            return false;
        }
        bool Join()
        {
            if (_joinable)
            {
                int n = pthread_join(_tid, nullptr);
                if (n != 0)
                    return false;
                _status = TSTATUS::STOP; // 一个线程能被join 那么这个线程的状态一定是STOP的
                return true;
            }
            return false;
        }
        void Detach() // 让线程分离
        {
            EnableDetach();
            pthread_detach(_tid);
        }
        bool IsJoinable() // 查一下线程是否 是可以被分离的
        {
            return _joinable;
        }

        std::string Name()
        {
            return _name;
        }
        ~Thread()
        {
        }

    private:
        std::string _name;
        pthread_t _tid;
        pid_t _pid;
        bool _joinable; // 判断是否是分离的，默认不是分离
        func_t _func;
        TSTATUS _status; // 表示线程自己的状态
    };
}

// #endif